Search CTRL + K

Learning Signed Distance Functions from Noisy 3D Point Clouds via Noise to Noise Mapping

Baorui Ma, Yu-Shen Liu, Zhizhong Han

ICML 2023 Oral

Tsinghua University

Wayne State University


Overview

Method Overview

  • N corrupted observations S={Ni|i[1,N]} of an uncorrupted 3D shape or scene S
  • Learn SDF f of S from S without ground truth signed distances, point normals, or clean point clouds
    • For a query location qR3 around S, train a neural network parameterized by θ to learn f

Noise2Noise Conclusion

Find the uncorrupted point cloud N from its corrupted observations NS below,

argminNEN{L(N,N)}
Transform into an equation with a denosing function F

argminFNiSNjSL(F(Ni),Nj).

Problems

  1. For 2D images, multiple corrupted observations have the pixel correspondence. This results in an assumption that all noisy observations at the same pixel location are ran- dom realizations of a distribution around a clean pixel value. However, this assumption is invalid for point clouds.
  2. How to learn SDF f via point cloud denoising

Why Define Denoising funcion F involving with SDF fθ


There are many potential paths to achieve this, but only one path is the shortest to the surface. Then involving with SDF, make the denosing function find the shortest path
For an arbitrary point n on the noisy point cloud N, define the denosing funtion below,

F(n,fθ)=nd×fθ(n,c)/fθ(n,c)2

where d=fθ(n,c)

Define the Distance Metric L

Conclusion : EMD is the better distance metric for comparing nosiy point cloud N and the underlying clean point cloud N

minθNiSNjSL(F(Ni,fθ),Nj)

  • (a) Nosiy Point Cloud
  • (b) Denosing Point Cloud by CD
  • (c) Denosing Point Cloud by EMD
  • (d) Reconstructed Mesh by MatchingCubes algorithm according the learned SDF (CD) without Geometric Consistency
  • (e) Reconstructed Mesh by MatchingCubes algorithm according the learned SDF (EMD) without Geometric Consistency
  • (f) Denosing Point Cloud by EMD with Geometric Consistency
  • (g) Reconstructed Mesh by MatchingCubes algorithm according the learned SDF (EMD) with Geometric Consistency
  • (h) Ground truth
Why (d) and (e) not good ?

Refer (b) and (c), minθNiSNjSL(F(Ni,fθ),Nj) constrains that points on the noisy point cloud should arrive onto the surface but there are no constraints on the paths to be the shortest. one situation that may happen is shown in (b), With the wrong signed distances f(θ) and gradient fθ, noises can also get pulled onto the surface, which results in a denoised point cloud with zero EMD distance to the clean point clouds. This is much different from the correct signed distance field that we expected in (c)

Geometric Consistency

for an arbitrary query n around a noisy point cloud Ni, the shortest distance between n and the surface can be either predicted by the SDF fθ or calculated based on the denosied point cloud Ni=F(Ni,fθ), both of which should be consistent to each other.

  • Therefore,the absolute value |fθ(n,c)| of the signed distance predicted at n should equal to the minimum distance between n and the denosied point cloud N=F(Ni,fθ), Since the point density of Ni may slightly affect the consistency, we leverage an inequality to describe the geometric consistency,
  • |fθ(n,c)|minnF(Ni,fθ)nn2
Loss Function

minθNiS(NjSL(F(Ni,fθ),Nj)+λ|Ni|nNiR(E)),

where

  • E=(|fθ(n,c)|minnF(Ni,fθ)nn2)
  • R(E)=max(0,E)
  • Comparing (c) and (f), with Geometric Consistency, the denoising Point Cloud are more uniformly distributed.
  • Comparing (e) and (g), learn a correct SDF to reconstruct plausible surface.

Experiments

(a)Visualization of optimization in 4 epochs via noise to noise mapping ; (b) Surface reconstruction and multiple level-sets


3 queries (black cubes) sampled from one noisy point cloud get pulled onto the surface

Point Cloud Denoising

  • Datasets: PU(PUNet) and PC(PointCleanNet)

Point Cloud Denoising

Point Cloud Upsampling

Surface Reconstruction for Shapes

Datasets: ShapeNet

Surface Reconstruction for Shapes

  • Dataset: FAMOUS and ABC.

Surface Reconstruction for Shapes

  • Dataset: D-FAUST and SRB


Surface Reconstruction for Scenes



Ablation Study

Effect of Density D of point cloud under PU

Effect of batch size B under PU

Effect of training steps

Effect of N under PU

  • More corrupted observations are the key to increase the performance of statistical reasoning although one corrupted observation is also fine to perform statistical reasoning well.
例子

例子

例子

例子

例子